# 初始化推理系统
from no_attention.inference import InferenceSystem

inference_system = InferenceSystem()

# 评估模型性能
results = inference_system.print_evaluation()

# 推理三个例子的主语和情绪
test_examples = [
    "这款手机的电池续航表现非常出色",
    "快递员的服务态度实在太差了",
    "餐厅的菜品质量令人惊艳",
    "快递员服务的还不错"
]

print("\n示例推理结果:")
print("=" * 50)
for example in test_examples:
    # 假设InferenceSystem类中有一个predict方法用于推理
    # 如果没有，需要在InferenceSystem类中添加该方法
    prediction = inference_system.predict(example)
    print(f"文本: {example}")
    print(f"主语: {prediction['subject']}")
    print(f"情绪: {prediction['sentiment']}")
    print("-" * 50)

# 也可以指定自定义测试数据集
# results = inference_system.print_evaluation("/Users/tony/work/python_workspace/train_min_model/no_attention/test_data.csv")